-
Notifications
You must be signed in to change notification settings - Fork 303
Add FLIC dataset #495
base: master
Are you sure you want to change the base?
Add FLIC dataset #495
Conversation
|
||
] | ||
|
||
flic_joint_label_names = [ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
How about flic_joint_names
?
the FLIC dataset. | ||
return_torsobox (bool): If :obj:`True`, this returns a bounding box | ||
around the torso. The default value is :obj:`False`. | ||
skip_bad (bool): If :obj:`True`, the data which have :obj:`isbad = 1` |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
How about use_bad
and use_unchecked
?
This style is consistent with the other datasets.
https://github.com/chainer/chainercv/blob/master/chainercv/datasets/voc/voc_bbox_dataset.py#L70
'isbad', | ||
'isunchecked', | ||
] | ||
labels = loadmat(os.path.join(data_dir, 'FLIC-full', 'examples.mat')) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
How about annos
?
labels
is overused in ChainerCV already.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Similarly, label_key
-> anno_key
.
Args: | ||
data_dir (string): Path to the root of the training data. If this is | ||
:obj:`auto`, this class will automatically download data for you | ||
under :obj:`$CHAINER_DATASET_ROOT/pfnet/chainercv/cub`. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
cub
-> flic
urls = [ | ||
'http://vision.grasp.upenn.edu/video/FLIC-full.zip', | ||
|
||
'http://cims.nyu.edu/~tompson/data/tr_plus_indices.mat', |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you delete this, or comment what this is?
data_root, | ||
ext) | ||
|
||
return data_root |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you make this function return
$CHAINER_ROOT/pfnet/chainercv/flic/FLIC-full
?
This is more consistent with other download functions.
Nice PR. |
@yuyu2172 Thanks for the review. I've fixed the points. |
Not sure what's blocking this PR, but just resolved conflicts for now. |
Please wait until #787 is merged |
This PR adds a dataset class for human pose dataset FLIC. It follows the convention of keypoint dataset like
CUBKeypointDataset
.